from django.db import models
# 自定义管理器
class BookManager(models.Manager):

    # 1.改写父类的方法
    def all(self):

        return super().filter(is_delete=True)

    # 2.新增自己的方法
    def create_book(self, name, pub_date):

        book = self.model()
        book.btitle = name
        book.bpub_date = pub_date
        book.bread = 0
        book.save()
        return book
#定义图书模型类BookInfo
class BookInfo(models.Model):

    """
    原因： 在models.py中定义了一个books = BookManager()管理器，相当于重写了一个管理器，所以objects管理器就不能用了。bookinfo这个类就没有objects这个属性。

解决办法：将自己建的管理器注释掉就可以了
    """
    # 关联新的管理器（老的不能用了）
    # books = BookManager()
    btitle = models.CharField(max_length=20, verbose_name='名称')
    bpub_date = models.DateField(verbose_name='发布日期')
    bread = models.IntegerField(default=0, verbose_name='阅读量')
    bcomment = models.IntegerField(default=0, verbose_name='评论量')
    is_delete = models.BooleanField(default=False, verbose_name='逻辑删除')

    # 添加图片字段
    # upload_to
    # null=True必须写，因为以前的老字段没有这一项
    image = models.ImageField(upload_to='auser',null=True,verbose_name="图片")
    class Meta:
        db_table = 'tb_books'  # 指明数据库表名
        verbose_name = '图书'  # 在admin站点中显示的名称
        verbose_name_plural = verbose_name  # 显示的复数名称

    def __str__(self):
        """定义每个数据对象的显示信息"""
        return self.btitle
    # 自定义方法列
    def my_date(self):

        return self.bpub_date.strftime('%Y-%m-%d')
    my_date.short_description = "我的日期"
    my_date.admin_order_field = "bpub_date"
#定义英雄模型类HeroInfo
class HeroInfo(models.Model):
    GENDER_CHOICES = (
        (0, 'female'),
        (1, 'male')
    )
    hname = models.CharField(max_length=20, verbose_name='名称')
    hgender = models.SmallIntegerField(choices=GENDER_CHOICES, default=0, verbose_name='性别')
    hcomment = models.CharField(max_length=200, null=True, verbose_name='描述信息')
    hbook = models.ForeignKey(BookInfo, on_delete=models.CASCADE, verbose_name='图书',related_name='hero')  # 外键
    is_delete = models.BooleanField(default=False, verbose_name='逻辑删除')

    class Meta:
        db_table = 'tb_heros'
        verbose_name = '英雄'
        verbose_name_plural = verbose_name

    def __str__(self):
        return self.hname
    # 自定义 阅读量的字段
    def read(self):

        return self.hbook.bread

    read.short_description = "书的阅读量"
